torch
causal-conv1d
mamba-ssm
timm
tensorboardX
einops
transformers
